package com.ruoyi.common.core.domain;

import com.ruoyi.common.utils.StringUtils;
import lombok.Data;

import java.io.Serial;
import java.util.HashMap;

/**
 * 长链接返回对象
 */
@Data
public class WebSocketResult extends HashMap<String, Object> {
    @Serial
    private static final long serialVersionUID = 1L;
    /**
     * 状态码
     */
    public static final String TYPE = "type";

    /**
     * 返回内容
     */
    public static final String MSG_TAG = "msg";

    /**
     * 数据对象
     */
    public static final String DATA_TAG = "data";

    /**
     * 初始化一个新创建的 WebSocketResult 对象
     *
     * @param type 类型
     * @param msg  返回内容
     */
    public WebSocketResult(String type, String msg) {
        super.put(TYPE, type);
        super.put(MSG_TAG, msg);
    }

    /**
     * 初始化一个新创建的 WebSocketResult 对象
     *
     * @param type 类型
     * @param msg  返回内容
     * @param data 数据对象
     */
    public WebSocketResult(String type, String msg, Object data) {
        super.put(TYPE, type);
        super.put(MSG_TAG, msg);
        if (StringUtils.isNotNull(data)) {
            super.put(DATA_TAG, data);
        }
    }


    /**
     * 返回成功消息
     *
     * @param msg 返回内容
     * @return 成功消息
     */
    public static WebSocketResult result(String type, String msg, Object data) {
        return new WebSocketResult(type, msg, data);
    }


    /**
     * 返回成功消息
     *
     * @param msg 返回内容
     * @return 成功消息
     */
    public static WebSocketResult result(String type, String msg) {
        return WebSocketResult.result(type, msg, null);
    }
}
